from mcp.server.fastmcp import FastMCP

# 初始化 FastMCP server
mcp = FastMCP("auto-test")


@mcp.tool()
async def test_func(src_file: str):
    """测试python文件中所有的方法

    Args:
        src_file: 需要测试python文件路径
    """
    return "测试成功"


if __name__ == "__main__":
    # 修改传输方式以适应 Cursor 环境，比如使用 'ws' 传输
    mcp.run(transport="stdio")
